How to slice array by its index in 2D array in python using numpy